#include "../mqserver/Messsage.hpp"
#include <gtest/gtest.h>
#include <iostream>
#include <unordered_map>
using namespace std;
mq::MessageManager::MessageManagerPtr mmp;
class MessageTest : public testing::Environment
{
public:
    virtual void SetUp() override
    {
        mmp = make_shared<mq::MessageManager>("./data");
        cout << "测试环境初始化" << endl;
        mmp->initQueueMessage("queue1");
        mmp->initQueueMessage("queue2");
    }
    virtual void TearDown() override
    {
        // mmp->clear();
        mmp->destroyQueueMessage("queue1");
        mmp->destroyQueueMessage("queue2");
        cout << "测试环境清理" << endl;
    }
};
// TEST(message, insert_test)
// {
//     {
//         mq::BasicProperties bp1;
//         bp1.set_id("1");
//         bp1.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp2;
//         bp2.set_id("2");
//         bp2.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp3;
//         bp3.set_id("3");
//         bp3.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp4;
//         bp4.set_id("4");
//         bp4.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp5;
//         bp5.set_id("5");
//         bp5.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mmp->insert("queue1", &bp1, "HelloWorld-1", true);
//         mmp->insert("queue1", &bp2, "HelloWorld-2", true);
//         mmp->insert("queue1", &bp3, "HelloWorld-3", true);
//         mmp->insert("queue1", &bp4, "HelloWorld-4", true);
//         mmp->insert("queue1", &bp5, "HelloWorld-5", false);
//         ASSERT_EQ(mmp->total_count("queue1"), 4);
//         ASSERT_EQ(mmp->getable_count("queue1"), 5);
//         ASSERT_EQ(mmp->durable_count("queue1"), 4);
//         ASSERT_EQ(mmp->waitack_count("queue1"), 0);
//     }

//     {
//         mq::BasicProperties bp1;
//         bp1.set_id("1");
//         bp1.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp2;
//         bp2.set_id("2");
//         bp2.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp3;
//         bp3.set_id("3");
//         bp3.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp4;
//         bp4.set_id("4");
//         bp4.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mq::BasicProperties bp5;
//         bp5.set_id("5");
//         bp5.set_delivery_mode(mq::DeliveryMode::DURABLE);
//         mmp->insert("queue2", &bp1, "HelloWorld-1", true);
//         mmp->insert("queue2", &bp2, "HelloWorld-2", false);
//         mmp->insert("queue2", &bp3, "HelloWorld-3", true);
//         mmp->insert("queue2", &bp4, "HelloWorld-4", true);
//         mmp->insert("queue2", &bp5, "HelloWorld-5", false);
//         ASSERT_EQ(mmp->total_count("queue2"), 3);
//         ASSERT_EQ(mmp->getable_count("queue2"), 5);
//         ASSERT_EQ(mmp->durable_count("queue2"), 3);
//         ASSERT_EQ(mmp->waitack_count("queue2"), 0);
//     }
// }

// TEST(exchange, select_test)
// {
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-1"));
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-2"));
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-3"));
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-4"));
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-5"));
//     ASSERT_EQ(mmp->total_count("queue1"), 4);
//     ASSERT_EQ(mmp->getable_count("queue1"), 0);
//     ASSERT_EQ(mmp->durable_count("queue1"), 4);
//     ASSERT_EQ(mmp->waitack_count("queue1"), 5);

//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-1"));
//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-2"));
//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-3"));
//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-4"));
//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-5"));
//     ASSERT_EQ(mmp->total_count("queue2"), 3);
//     ASSERT_EQ(mmp->getable_count("queue2"), 0);
//     ASSERT_EQ(mmp->durable_count("queue2"), 3);
//     ASSERT_EQ(mmp->waitack_count("queue2"), 5);
// }
// TEST(exchange, ack_test)
// {
//     mmp->ack("queue1", "1");
//     mmp->ack("queue1", "3");
//     ASSERT_EQ(mmp->total_count("queue1"), 4);
//     ASSERT_EQ(mmp->getable_count("queue1"), 0);
//     ASSERT_EQ(mmp->durable_count("queue1"), 2);
//     ASSERT_EQ(mmp->waitack_count("queue1"), 3);

//     mmp->ack("queue2", "1");
//     mmp->ack("queue2", "2");
//     mmp->ack("queue2", "3");
//     ASSERT_EQ(mmp->total_count("queue2"), 3);
//     ASSERT_EQ(mmp->getable_count("queue2"), 0);
//     ASSERT_EQ(mmp->durable_count("queue2"), 1);
//     ASSERT_EQ(mmp->waitack_count("queue2"), 2);
// }
// TEST(message, recovery_test)
// {
//     ASSERT_EQ(mmp->total_count("queue1"), 2);
//     ASSERT_EQ(mmp->getable_count("queue1"), 2);
//     ASSERT_EQ(mmp->durable_count("queue1"), 2);
//     ASSERT_EQ(mmp->waitack_count("queue1"), 0);
//     ASSERT_EQ(mmp->total_count("queue2"), 1);
//     ASSERT_EQ(mmp->getable_count("queue2"), 1);
//     ASSERT_EQ(mmp->durable_count("queue2"), 1);
//     ASSERT_EQ(mmp->waitack_count("queue2"), 0);

//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-2"));
//     ASSERT_EQ(mmp->front("queue1")->payload().body(), string("HelloWorld-4"));
//     ASSERT_EQ(mmp->front("queue1").get(), nullptr);

//     ASSERT_EQ(mmp->front("queue2")->payload().body(), string("HelloWorld-4"));
//     ASSERT_EQ(mmp->front("queue2").get(), nullptr);
// }
int main(int argc, char *argv[])
{
    testing::AddGlobalTestEnvironment(new MessageTest);
    testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}